#!/usr/bin/env python
#-*- encoding:utf-8 -*-
# from 豆包 AI
# 功能：将t1.txt的格式转换为pyPhrase.txt(搜狗新版词库)风格，并输出到a.org
# 核心逻辑：1. 读取t1.txt的制表符分隔内容 2. 将拼音的空格替换为单引号 3. 重组为“处理后拼音 中文”格式 4. 写入目标文件

def convert_t1_to_pyphrase_style(input_path="t.txt", output_path="a.org"):
    try:
        # 1. 读取t1.txt文件（指定UTF-8编码避免中文乱码）
        with open(input_path, "r", encoding="utf-8") as input_file:
            # 读取所有行并去除空行/无效行
            raw_lines = [line.strip() for line in input_file if line.strip()]
        
        # 2. 处理每一行的格式
        processed_lines = []
        for idx, line in enumerate(raw_lines, 1):
            # 按制表符分割（t1.txt格式：中文\t拼音\t1）
            line_parts = line.split("\t")
            
            # 跳过格式异常的行（避免脚本崩溃）
            if len(line_parts) != 3:
                print(f"警告：第{idx}行格式异常（非3段），已跳过 -> 内容：{line}")
                continue
            
            # 提取中文、拼音（忽略最后一位的数字“1”）
            chinese_content = line_parts[0]
            pinyin_content = line_parts[1]
            
            # 将拼音的“空格分隔”转为“单引号分隔”（匹配pyPhrase.txt风格）
            processed_pinyin = pinyin_content.replace(" ", "'")
            
            # 重组为“处理后拼音 中文”格式（与pyPhrase.txt的“a'a 啊啊”一致）
            new_line = f"{processed_pinyin} {chinese_content}"
            processed_lines.append(new_line)
        
        # 3. 将处理后的内容写入a.org文件
        with open(output_path, "w", encoding="utf-8") as output_file:
            # 每行结尾添加换行符，确保文件格式规范
            output_file.write("\n".join(processed_lines))
        
        print(f"格式转换完成！共处理{len(processed_lines)}行有效数据，结果已保存到：{output_path}")
    
    except FileNotFoundError:
        print(f"错误：输入文件 {input_path} 未找到，请检查文件路径是否正确！")
    except Exception as e:
        print(f"脚本运行出错：{str(e)}")

# 执行转换函数（默认读取t1.txt，输出到a.org）
if __name__ == "__main__":
    convert_t1_to_pyphrase_style()
